package com.leetcode.problems_10;

import java.util.Arrays;

/**
 * 0028: 实现strStr
 * 实现strStr()函数。
 *
 * 给你两个字符串haystack 和 needle，请你在 haystack 字符串中找出 needle 字符串出现的第一个位置（下标从 0 开始）。
 * 如果不存在，则返回 -1 。
 *
 * 说明：
 *      当needle是空字符串时，我们应当返回什么值呢？这是一个在面试中很好的问题。
 *      对于本题而言，当needle是空字符串时我们应当返回 0 。
 *      这与 C 语言的strstr()以及 Java 的indexOf()定义相符。
 *
 * 示例 1：
 *      输入：haystack = "hello", needle = "ll"
 *      输出：2
 *
 * 示例 2：
 *      输入：haystack = "aaaaa", needle = "bba"
 *      输出：-1
 *
 * 示例 3：
 *      输入：haystack = "", needle = ""
 *      输出：0
 *
 * tips：KMP算法
 *      注意next数组的求解
 *          前缀表
 *          前缀表-1
 *
 * @Author: WuPeng
 * @Date: 2021/5/19 2:55 下午
 */
public class Strstr {

    public static int solution(String haystack, String needle){
        if (needle.length() == 0) {
            return 0;
        }
        int[] next = getNext(needle);
        int j = -1;

        for (int i = 0; i < haystack.length(); i++) {

            // 不匹配时，j到next数组中寻找之前匹配的位置
            while (j >= 0 && haystack.charAt(i) != needle.charAt(j+1)){
                j = next[j];
            }

            // 匹配时，j和i同时向后移动
            if (haystack.charAt(i) == needle.charAt(j+1)){
                j++;
            }

            // 文本串haystack中出现了模式串
            if (j == (needle.length() - 1)){
                return (i-needle.length() + 1);
            }
        }
        return -1;
    }

    private static int[] getNext(String str){
        int[] next = new int[str.length()];
        int j = -1;
        next[0] = j;
        for (int i = 1; i < str.length(); i++) {
            // 前后缀不匹配时
            while (j >= 0 && str.charAt(i) != str.charAt(j+1)){
                j = next[j];
            }
            // 前后缀匹配时
            if (str.charAt(i) == str.charAt(j+1)){
                j++;
            }
            next[i] = j;
        }
        return next;
    }

    public static void main(String[] args) {
        String haystack = "aabaabaaf";
        String needle = "aabaaf";
        int[] next = getNext(needle);
//        System.out.println(Arrays.toString(next));
        System.out.println(solution(haystack, needle));
    }

}
